export MODEL_DIR="../../../../stable-diffusion-v1-5-text"
export OUTPUT_DIR="./models_japan_cartoon"

export TORCH_USE_CUDA_DSA=0
export CUDA_LAUNCH_BLOCKING=1

accelerate launch train_text_to_image_lora_lth.py \
 --pretrained_model_name_or_path=$MODEL_DIR \
 --output_dir=$OUTPUT_DIR \
 --dataset_name=../controlnet/a34611c51dcbae2e7e4863c9143f18814df8d43400872ac43c3a4509f1da8d53 \
 --resolution=256 \
 --learning_rate=1e-5 \
 --validation_prompt="japan cartoon style, a girl with a hat and a flower" \
 --train_batch_size=1 \
 --gradient_accumulation_steps=1 \
 --gradient_checkpointing \
 --use_8bit_adam \
 --checkpoints_total_limit 2 \
 --resolution 128 \
 --mixed_precision no
